package net.tv90.bfmccore.slot;

import lombok.Setter;

import java.util.HashMap;

public class Attachment {
    private @Setter HashMap<EnumAttachmentType, String> attachmentType = new HashMap<>();

    public Attachment() {
    }

    public void addAttachment(EnumAttachmentType enumAttachmentType, String itemID) {
        attachmentType.put(enumAttachmentType, itemID);
    }

    public HashMap<String, String> getStringMap() {
        HashMap<String, String> map = new HashMap<>();
        attachmentType.forEach((K, V) -> {
            map.put(K.name(), V);
        });
        return map;
    }

    public enum EnumAttachmentType {
        scope, barrel, grip, gadget, pump,
    }
}
